import torch


model = torch.hub.load('C:/Users/Administrator/Desktop/游戏自动化/cs/yolov5', 'custom', 'C:/Users/Administrator/Desktop/游戏自动化/cs/yolov5/yolov5s.pt',source='local', force_reload=False) 
img = "data/images/bus.jpg"
reslut = model(img)
reslut.save("predict.jpg")